# ---------------------------------------------------------------
# Copyright (c) 2021-2022 ETH Zurich, Lukas Hoyer. All rights reserved.
# Licensed under the Apache License, Version 2.0
# ---------------------------------------------------------------

# Baseline UDA
uda = dict(
    type='MultiTeacherIMDTGT',
    alpha=0.99,
    pseudo_threshold=0.968,
    pseudo_weight_ignore_top=200,
    pseudo_weight_ignore_bottom=0,
    pseudo_ref_weight_ignore_top = 150,
    use_ref_label = True,
    mix='class',
    blur=True,
    color_jitter_strength=0.6,
    color_jitter_probability=0.2,
    debug_img_interval=999,
    num_classes=6,  # 假设你的数据集有6个类别
)
use_ddp_wrapper = True
